shared source/wpf/src/host/shimimpl/persisthistory.cxx

   1 //+-----------------------------------------------------------------------
   2 //
   3 //  Copyright (c) Microsoft Corporation.  All rights reserved.
   4 //
   5 //  Description:
   6 //     Defines the PersistHistory class of PresentationHost.
   7 //
   8 //  History:
   9 //     2002/06/19-murrayw
  10 //          Created
  11 //     2007/09/20-[....]
  12 //          Ported Windows->DevDiv. See SourcesHistory.txt.
  13 //
  14 //------------------------------------------------------------------------
  15
  16 #include "Precompiled.hxx"
  17 #include "PersistHistory.hxx"
  18 #include "HostShim.hxx"
  19 #include "Version.hxx"
  20
  21 // This macro will keep the types in [....] between the LoadHistoryHelper and SaveHistoryHelper methods.
  22 #define STARTUP_URI_LENGTH_TYPE ULONG
  23
  24 /******************************************************************************
  25  CPersistHistory::CPersistHistory()
  26 ******************************************************************************/
  27 CPersistHistory::CPersistHistory(CHostShim *pHostShim)
  28 {
  29     m_pHostShim = pHostShim;
  30 }
  31
  32 /******************************************************************************
  33  CPersistHistory::~CPersistHistory()
  34 ******************************************************************************/
  35 CPersistHistory::~CPersistHistory()
  36 {
  37 }
  38
  39 /******************************************************************************
  40  CPersistHistory::QueryInterface()
  41 ******************************************************************************/
  42 STDMETHODIMP CPersistHistory::QueryInterface(REFIID riid, LPVOID *ppReturn)
  43 {
  44     return m_pHostShim->QueryInterface(riid, ppReturn);
  45 }
  46
  47 /******************************************************************************
  48  CPersistHistory::AddRef()
  49 ******************************************************************************/
  50 STDMETHODIMP_(ULONG) CPersistHistory::AddRef()
  51 {
  52     return m_pHostShim->AddRef();
  53 }
  54
  55 /******************************************************************************
  56   CPersistHistory::Release()
  57 ******************************************************************************/
  58 STDMETHODIMP_(ULONG) CPersistHistory::Release()
  59 {
  60     return m_pHostShim->Release();
  61 }
  62
  63 /******************************************************************************
  64  CPersistHistory::GetClassID()
  65 ******************************************************************************/
  66 STDMETHODIMP CPersistHistory::GetClassID(LPCLSID pClassID)
  67 {
  68     return m_pHostShim->GetClassID(pClassID);
  69 }
  70
  71 /******************************************************************************
  72  CPersistHistory::SetPositionCookie
  73 ******************************************************************************/
  74 STDMETHODIMP CPersistHistory::SetPositionCookie(DWORD /*dwPositionCookie*/)
  75 {
  76     return S_OK;
  77 }
  78
  79 /******************************************************************************
  80  CPersistHistory::GetPositionCookie
  81 ******************************************************************************/
  82 STDMETHODIMP CPersistHistory::GetPositionCookie(__out DWORD* pdwPositionCookie)
  83 {
  84     HRESULT hr = S_OK;
  85
  86     CK_PARG(pdwPositionCookie);
  87
  88     *pdwPositionCookie = 0;
  89
  90 Cleanup:
  91
  92     return hr;
  93 }
  94
  95 /**************************************************************************
  96    CPersistHistory::SaveHistory(IStream *pStream)
  97 **************************************************************************/
  98 STDMETHODIMP CPersistHistory::SaveHistory(__in IStream* pSaveStream)
  99 {
 100     HRESULT hr = S_OK;
 101
 102     CKHR(m_pHostShim->RequestedVersion()->WriteToStream(pSaveStream));
 103     CKHR(WriteToStream(pSaveStream, (DWORD) m_pHostShim->GetMimeType()));
 104     CKHR(m_pHostShim->StartupUri()->WriteToStream(pSaveStream));
 105     CKHR(m_pHostShim->StartupLocation()->WriteToStream(pSaveStream));
 106     CKHR(m_pHostShim->ApplicationIdentity()->WriteToStream(pSaveStream));
 107
 108     if(!m_pHostShim->GetVersion())
 109     {
 110         CKHR(E_UNEXPECTED);
 111     }
 112     CKHR(m_pHostShim->GetVersion()->SaveToHistory(pSaveStream));
 113
 114 Cleanup:
 115     return hr;
 116 }
 117
 118 /**************************************************************************
 119    CPersistHistory::LoadHistory(IStream *pStream, IBindCtx *pbc)
 120 **************************************************************************/
 121 STDMETHODIMP CPersistHistory::LoadHistory(__in IStream* pLoadStream, IBindCtx* pBindCtx)
 122 {
 123     HRESULT hr = S_OK;
 124
 125     // When doing navigations inside of an app, history is saved without going through
 126     // CPersistHistory::SaveHistory, so the information that puts in the stream isn't there.
 127     // This check is to be sure that we are being activated by IPersistHistory.
 128
 129     if (!m_pHostShim->GetVersion())
 130     {
 131         CKHR(m_pHostShim->RequestedVersion()->ReadFromStream(pLoadStream));
 132
 133         DWORD mimeType;
 134         CKHR(ReadFromStream(pLoadStream, &mimeType));
 135         m_pHostShim->SetMimeType((MimeType)mimeType);
 136
 137         CKHR(m_pHostShim->StartupUri()->ReadFromStream(pLoadStream));
 138         CKHR(m_pHostShim->StartupLocation()->ReadFromStream(pLoadStream));
 139         CKHR(m_pHostShim->ApplicationIdentity()->ReadFromStream(pLoadStream));
 140
 141         pLoadStream->AddRef();
 142         m_pHostShim->SetHistoryStream(pLoadStream);
 143
 144         m_pHostShim->Execute(); //[Watson on error]
 145     }
 146
 147     CKHR(m_pHostShim->GetVersion()->LoadFromHistory(pLoadStream, pBindCtx));
 148
 149 Cleanup:
 150     return hr;
 151 }
 152
 153 STDMETHODIMP CPersistHistory::ReadFromStream(IStream* pInputStream, DWORD* pdwData)
 154 {
 155     HRESULT hr = S_OK;
 156
 157     CHECK_POINTER(pdwData);
 158
 159     ULONG bytesRead = 0;
 160     CKHR(pInputStream->Read((void*)pdwData, sizeof(*pdwData), &bytesRead));
 161
 162 Cleanup:
 163     return hr;
 164 }
 165
 166 STDMETHODIMP CPersistHistory::WriteToStream(IStream* pOutputStream, DWORD dwData)
 167 {
 168     HRESULT hr = S_OK;
 169
 170     ULONG bytesWritten = 0;
 171     CKHR(pOutputStream->Write((void*)&dwData, sizeof(dwData), &bytesWritten));
 172
 173 Cleanup:
 174     return hr;
 175 }